"""
数据库模型模块
包含所有数据库表的模型定义
"""

# 导入所有模型，确保它们被正确注册
from .user import User
from .user_settings import UserSettings
from .conversation import Conversation, Message
from .model_config import ModelConfig
from .prompt_template import PromptTemplate
from .knowledge_base import KnowledgeBase, Document
from .image_generation import ImageGeneration
from .mcp_session import MCPSession
from .mcp_context import MCPContext
from .mcp_message import MCPMessage

__all__ = [
    "User",
    "UserSettings", 
    "Conversation",
    "Message",
    "ModelConfig",
    "PromptTemplate",
    "KnowledgeBase",
    "Document",
    "ImageGeneration",
    "MCPSession",
    "MCPContext",
    "MCPMessage"
]
